class Config:
    # 数据配置
    NUM_USERS = 100000
    NUM_ITEMS = 50000
    EMBEDDING_DIM = 64
    POSITIVE_SAMPLES_PER_USER = 10
    NEGATIVE_SAMPLES_PER_POSITIVE = 4
    
    # 训练配置
    BATCH_SIZE = 4096
    EPOCHS = 5
    LEARNING_RATE = 0.001
    TEST_SIZE = 0.2
    TOWERS_MODEL_SAVE_PATH = "model/two_tower_model.pth"
    DNN_MODEL_SAVE_PATH = "model/dnn_model.pth"
    
    # 索引配置
    INDEX_PATH = "index/faiss_index.bin"
    INDEX_TYPE = "FlatIP"  # 可选 IVFFlat, HNSW 等
    
    # API配置
    API_HOST = "0.0.0.0"
    API_PORT = 5000
    TOP_K = 5